# main.py
from data_manager import StockDataManager

def main():
    # 初始化数据管理器
    manager = StockDataManager()

    # 1. 更新股票列表
    print("正在更新股票列表...")
    manager.update_stock_list('akshare')

    # 2. 更新日线数据（可以指定部分股票测试）
    test_symbols = ['000001', '000002', '600036', '601318']
    print("正在更新日线数据...")
    manager.update_daily_data(test_symbols, 'akshare')

    # 3. 查询数据
    print("查询平安银行数据...")
    df = manager.get_data('000001', '2023-01-01', '2023-12-31')
    print(f"获取到 {len(df)} 条记录")
    print(df.head())

if __name__ == "__main__":
    main()